import argparse
    
parser = argparse.ArgumentParser()
parser.add_argument("--sourcePath", default="data/sourceData", type=str)
parser.add_argument("--processedPath", default="data/processedData", type=str)
parser.add_argument("--savePath", default="result", type=str)


parser.add_argument("--learning_rate", default=1e-3, type=float)
parser.add_argument("--batch_size", default=32, type=int)
parser.add_argument("--num_epochs", default=10, type=int)

parser.add_argument("action", type=str, choices=["clearnForTrain", "doTrain", "clearnForPred", "doPred", "update","doPlot"])
        
parser.add_argument("--trainFn", type=str)
parser.add_argument("--devFn", type=str)
parser.add_argument("--predFn", type=str)

args = parser.parse_args()

if args.action=="clearnForPred" and args.predFn is None:
    parser.error('预测操作需指定预测文件')
